<%@ page language="java" contentType="text/html; charset=UTF-8" pageEncoding="UTF-8"%>
<%@ page import="java.util.*,java.io.*" %>
<%@ page import="java.text.SimpleDateFormat" %>
<%@ page import="org.apache.commons.fileupload.*" %>
<%@ page import="org.apache.commons.fileupload.disk.*" %>
<%@ page import="org.apache.commons.fileupload.servlet.*" %>
<%@ page import="org.json.simple.*" %>
<%@ page import="com.db.SharePassageDao" %>
<%



	//文件保存目录路径
    String savePath = pageContext.getServletContext().getRealPath("/") + "psgPhoto/";
	String savePath2 = "D:/jee-neon/workspace/traveling/WebRoot/psgPhoto/image/image/";
    System.out.println(pageContext.getServletContext().getRealPath("/"));
    //文件保存目录URL
    String saveUrl  = request.getContextPath() + "/psgPhoto/";
    //String saveUrl  = "/../../psgPhoto/";
    //String saveUrl  = "D:/jee-neon/workspace/traveling/WebRoot/psgPhoto/";




//定义允许上传的文件扩展名
	HashMap<String, String> extMap = new HashMap<String, String>();
	extMap.put("image", "gif,jpg,jpeg,png,bmp");
	extMap.put("flash", "swf,flv");
	extMap.put("media", "swf,flv,mp3,wav,wma,wmv,mid,avi,mpg,asf,rm,rmvb");
	extMap.put("file", "doc,docx,xls,xlsx,ppt,htm,html,txt,zip,rar,gz,bz2");

//最大文件大小
	long maxSize = 1000000;

	response.setContentType("text/html; charset=UTF-8");

	if(!ServletFileUpload.isMultipartContent(request)){
		out.println(getError("请选择文件。"));
		return;
	}

//检查目录
	File uploadDir = new File(savePath);
	if(!uploadDir.isDirectory()){
		out.println(getError("上传目录不存在。"));
		return;
	}
//检查目录写权限
	if(!uploadDir.canWrite()){
		out.println(getError("上传目录没有写权限。"));
		return;
	}

	String dirName = request.getParameter("dir");
	if (dirName == null) {
		dirName = "image";
	}
	if(!extMap.containsKey(dirName)){
		out.println(getError("目录名不正确。"));
		return;
	}
//创建文件夹
	savePath += dirName + "/";
	saveUrl += dirName + "/";
	File saveDirFile = new File(savePath);
	if (!saveDirFile.exists()) {
		saveDirFile.mkdirs();
	}
	SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
	String ymd = "image";
	savePath += ymd + "/";
	saveUrl += ymd + "/";
	File dirFile = new File(savePath);
	out.print(dirFile);
	if (!dirFile.exists()) {
		dirFile.mkdirs();
	}

	FileItemFactory factory = new DiskFileItemFactory();
	ServletFileUpload upload = new ServletFileUpload(factory);
	upload.setHeaderEncoding("UTF-8");
	List items = upload.parseRequest(request);
	Iterator itr = items.iterator();
	while (itr.hasNext()) {
		FileItem item = (FileItem) itr.next();
		String fileName = item.getName();
		long fileSize = item.getSize();
		if (!item.isFormField()) {
			//检查文件大小
			if(item.getSize() > maxSize){
				out.println(getError("上传文件大小超过限制。"));
				return;
			}
			//检查扩展名
			String fileExt = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase();
			if(!Arrays.<String>asList(extMap.get(dirName).split(",")).contains(fileExt)){
				out.println(getError("上传文件扩展名是不允许的扩展名。\n只允许" + extMap.get(dirName) + "格式。"));
				return;
			}
			/*************************************文件名重命名***************************************/
			SharePassageDao connectPassageForID = new SharePassageDao();
			final int passageID = connectPassageForID.searchPassageID()+1;
			/*************************************判断文件名是否存在*********************************/
			class FindFile{

				public int getNumber(){

					//创建fileFolder对象，指向文件路径
					String fileFolder = "D:/jee-neon/workspace/traveling/WebRoot/psgPhoto/image/image/";
					String file_Name;
					//遍历所有文件名称将最新的下标值加1
					File fd = new File(fileFolder);
					//文件名数组
					File fl[] = fd.listFiles();
					List<File> Lfs = new ArrayList<File>();
					int last=0,getPassage=0;
					int begin = 0,end = 0,getIndex = 0;
					int fileNum = 0;//文件夹下的文件初始化
					int maxIndex=0;

					//读取文件名增加下标
					//判断是否存在有passageID的文件名
					for(int i=0;i<fl.length;i++)
					{
						File fs = fl[i];
						if(fs.isFile())
						{
							//获得文件的前缀
							fileNum++;
							file_Name = fs.getName();
							last=file_Name.indexOf("_");
							getPassage=Integer.parseInt(file_Name.substring(0,last));

							//第i个文件的前缀是passageID，需要判断他的后缀的值
							if (getPassage==passageID)
							{
								//获得文件的后缀
								begin = file_Name.lastIndexOf("_");
								end = file_Name.indexOf(".");
								getIndex = Integer.parseInt(file_Name.substring(begin+1,end));

								//获得后缀的最大值
								if (getIndex>maxIndex){
									maxIndex=getIndex;

								}
								else;
							}
							else;
						}

					}
					//所有循环结束，不存在以passageID开头的文件
					return maxIndex+1;
				}
			}
			FindFile getPassageNumber=new FindFile();
			int lastNumber=getPassageNumber.getNumber();

			/*************************************判断文件名是否存在**********************************/

			String newFileName = passageID + "_" + lastNumber + "." + fileExt;
			try{
				File uploadedFile = new File(savePath, newFileName);
				File uploadedFile2 = new File(savePath2, newFileName);

				item.write(uploadedFile);
				item.write(uploadedFile2);
			}catch(Exception e){
				out.println(getError("上传文件失败。"));
				return;
			}
			/*************************************文件名重命名***************************************/


			JSONObject obj = new JSONObject();
			obj.put("error", 0);
			obj.put("url", saveUrl + newFileName);
			out.println(obj.toJSONString());
			System.out.println(savePath);
			System.out.println(saveUrl);
			
		}
	}
%>
<%!
	private String getError(String message) {
		JSONObject obj = new JSONObject();
		obj.put("error", 1);
		obj.put("message", message);
		return obj.toJSONString();
	}
%>